#include <stdexcept>
#include <iostream>
#include "../common/PrintUtil.h"

using namespace std;

/* 基于邻接矩阵实现的无向图类 */
class GraphAdjMat {
    vector<int> vertices; 
    vector<vector<int>> adjMat;

public:
    /* 构造方法 */
    GraphAdjMat(const vector<int>& vertices, const vector<vector<int>>& edges) {
        for (int val : vertices) {
            addVertex(val);
        }
        for (const vector<int>& edge : edges) {
            addEdge(edge[0], edge[1]);
        }
    }

    /* 获取顶点数量 */
    int size() const {
        return vertices.size();
    }

    /* 添加顶点 */
    void addVertex(int val) {
        int n = size();
        vertices.push_back(val);
        adjMat.emplace_back(n, 0);
        for (vector<int>& row : adjMat) {
            row.push_back(0);
        }
    }

    /* 删除顶点 */
    void removeVertex(int index) {
        if (index >= size()) {
            throw out_of_range("顶点不存在");
        }
        vertices.erase(vertices.begin() + index);
        adjMat.erase(adjMat.begin() + index);
        for (vector<int>& row : adjMat) {
            row.erase(row.begin() + index);
        }
    }

    /* 添加边 */
    // 参数 i, j 对应 vertices 元素索引
    void addEdge(int i, int j) {
        if (i < 0 || j < 0 || i >= size() || j >= size() || i == j) {
            throw out_of_range("顶点不存在");
        }
        adjMat[i][j] = 1;
        adjMat[j][i] = 1;
    }

    /* 删除边 */
    // 参数 i, j 对应 vertices 元素索引
    void removeEdge(int i, int j) {
        if (i < 0 || j < 0 || i >= size() || j >= size() || i == j) {
            throw out_of_range("顶点不存在");
        }
        adjMat[i][j] = 0;
        adjMat[j][i] = 0;
    }

    /* 打印邻接矩阵 */
    void print() {
        cout << "顶点列表 = ";
        PrintUtil::printVector(vertices);
        cout << "邻接矩阵 =" << endl;
        PrintUtil::printVectorMatrix(adjMat);
    }
};
